package com.lucky.carcodepro.config.mybatis;

import cn.hutool.core.util.ObjUtil;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.BlockAttackInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.OptimisticLockerInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.lucky.carcodepro.common.CurrentContextHolder;
import com.lucky.carcodepro.domain.bo.CurrentContextBo;
import com.lucky.carcodepro.domain.model.base.BaseModel;
import com.lucky.carcodepro.domain.model.base.FlowModel;
import com.lucky.carcodepro.domain.model.base.PackageModel;
import com.lucky.carcodepro.utils.DateUtils;
import org.apache.ibatis.reflection.MetaObject;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.transaction.annotation.EnableTransactionManagement;

import java.time.LocalDateTime;

/**
 * MybatisPlusConfig
 * <p>
 * 描述：MP自定义配置类
 *
 * @author Annie
 * @version 2024/10/2 16:09
 */
@Configuration
@EnableTransactionManagement(proxyTargetClass = true)
@MapperScan(basePackages = "com.lucky.carcodepro.mapper")
public class MybatisPlusConfig implements MetaObjectHandler
{

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor()
    {

        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        // 分页插件
        interceptor.addInnerInterceptor(paginationInnerInterceptor());
        // 乐观锁插件
        interceptor.addInnerInterceptor(optimisticLockerInnerInterceptor());
        // 阻断插件
        interceptor.addInnerInterceptor(blockAttackInnerInterceptor());
        // 数据库类型
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
        return interceptor;
    }

    /**
     * 分表插件
     *
     */
    @Bean
    public SimpleShardingInterceptor mySimpleShardingPlugin()
    {
        return new SimpleShardingInterceptor();
    }


    @Override
    public void insertFill(MetaObject metaObject)
    {
        this.setFieldValByName(BaseModel.Fields.createTime, LocalDateTime.now(), metaObject);
        this.setFieldValByName(BaseModel.Fields.updateTime, LocalDateTime.now(), metaObject);
        if (metaObject.getOriginalObject() instanceof PackageModel)
        {
            // 包 ID 注入
            CurrentContextBo currentContextBo = CurrentContextHolder.get();
            Integer packageId = 1;
            if (ObjUtil.isNotNull(currentContextBo) && ObjUtil.isNotNull(currentContextBo.getUser()))
            {
                packageId = currentContextBo.getUser().getPackageId();
            }
            this.setFieldValByName(PackageModel.Fields.packageId, packageId, metaObject);
        }
        if (metaObject.getOriginalObject() instanceof FlowModel)
        {
            // 操作者模式 注入
            String nowMonthDay = DateUtils.getNowMonthDay();
            this.setFieldValByName(FlowModel.Fields.createDate, nowMonthDay, metaObject);
            this.setFieldValByName(FlowModel.Fields.updateDate, nowMonthDay, metaObject);
        }

    }

    @Override
    public void updateFill(MetaObject metaObject)
    {
        this.setFieldValByName(BaseModel.Fields.updateTime, LocalDateTime.now(), metaObject);
        if (metaObject.getOriginalObject() instanceof FlowModel)
        {
            String nowMonthDay = DateUtils.getNowMonthDay();
            this.setFieldValByName(FlowModel.Fields.updateDate, nowMonthDay, metaObject);
        }
    }


    /**
     * 分页插件，自动识别数据库类型 https://baomidou.com/guide/interceptor-pagination.html
     */
    public PaginationInnerInterceptor paginationInnerInterceptor()
    {
        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
        // 设置数据库类型为mysql
        paginationInnerInterceptor.setDbType(DbType.MYSQL);
        // 设置最大单页限制数量，默认 500 条，-1 不受限制
        paginationInnerInterceptor.setMaxLimit(-1L);
        return paginationInnerInterceptor;
    }

    /**
     * 乐观锁插件 https://baomidou.com/guide/interceptor-optimistic-locker.html
     */
    public OptimisticLockerInnerInterceptor optimisticLockerInnerInterceptor()
    {
        return new OptimisticLockerInnerInterceptor();
    }

    /**
     * 如果是对全表的删除或更新操作，就会终止该操作
     * https://baomidou.com/guide/interceptor-block-attack.html
     */
    public BlockAttackInnerInterceptor blockAttackInnerInterceptor()
    {
        return new BlockAttackInnerInterceptor();
    }

}
